import torch
import torch.nn as nn

# 创建一个卷积层
conv_layer = nn.Conv2d(1, 6, kernel_size=5)

# 打印卷积核的初始权重
print("卷积核初始权重:")
print(conv_layer.weight)